import { useMutation } from 'react-query';
import { sendData, sendDataConfirm, sendPresignedURL } from './api';

interface UseMutateSendDataProps {
  headerImageFile: File;
  coreValueImageFiles: File[];
  memberImageFiles: File[];
  recruitHeaderImageFile: File;
}

const useMutateSendData = ({
  headerImageFile,
  coreValueImageFiles,
  memberImageFiles,
  recruitHeaderImageFile,
}: UseMutateSendDataProps) => {
  const { mutate: sendMutate, isLoading: sendIsLoading } = useMutation({
    mutationFn: (data) => sendData(data),
    onSuccess: async (res: any) => {
      const {
        generation,
        coreValues,
        headerImage: headerImageURL,
        members,
        recruitHeaderImage: recruitHeaderImageURL,
      } = res.data;

      try {
        const imageUploadTasks = [
          { url: headerImageURL, file: headerImageFile },
          ...coreValues.map((value, index) => ({
            url: value.image,
            file: coreValueImageFiles[index],
          })),
          ...members.map((member, index) => ({
            url: member.profileImage,
            file: memberImageFiles[index],
          })),
          { url: recruitHeaderImageURL, file: recruitHeaderImageFile },
        ];

        await Promise.all(
          imageUploadTasks.map(({ url, file }) =>
            sendPresignedURL(url, file).catch((error) => {
              console.error(`Failed to upload ${file.name}:`, error);
              throw error;
            }),
          ),
        );

        const finalResponse = await sendDataConfirm({ generation });
        return finalResponse;
      } catch (error) {
        console.error('Error during file upload or confirmation:', error);
        throw error;
      }
    },
  });

  return { sendMutate, sendIsLoading };
};

export default useMutateSendData;